package com.Tradition.controller.sys;

import com.Tradition.common.aop.annotation.Di;
import com.Tradition.common.http.VerificationCodeEnum;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.lang.reflect.Field;
import java.util.HashSet;
import java.util.Set;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;

/**
 * VerificationCode
 *
 * @author 朱笑
 * @date 2018-06-20 16:53
 */
@Api(value="获取验证码",tags={"获取验证码"})
@Controller
public class VerificationCodeController {

    static Set<String> codeNameSet=new HashSet<>();
    static {
        Field[] Fields=VerificationCodeEnum.class.getFields();
        for(Field item:Fields){
            codeNameSet.add(item.getName());
        }
    }

    /*实现的核心代码*/
    @ApiOperation(value="用户登录接口",notes="验证码：LOGIN,REG")
    @GetMapping("VerificationCode.do")
    @Di()
    public  void doPost(@ApiParam(name = "验证码类型", value = "codeName", required = true) String codeName, HttpServletRequest request, HttpServletResponse response)
            throws IOException {
        if(!checkName(codeName)){//不属于系统验证码
            response.getWriter().write("change GAY");
            response.getWriter().close();
            return ;
        }
        response.setContentType("image/jpeg");
        HttpSession session=request.getSession();
        int width=60;
        int height=20;

        //设置浏览器不要缓存此图片
        response.setHeader("Pragma", "No-cache");
        response.setHeader("Cache-Control", "no-cache");
        response.setDateHeader("Expires", 0);

        //创建内存图像并获得图形上下文
        BufferedImage image=new BufferedImage(width, height,BufferedImage.TYPE_INT_RGB);
        Graphics g=image.getGraphics();

        /*
         * 产生随机验证码
         * 定义验证码的字符表
         */
        String chars="0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
        char[] rands=new char[4];
        for(int i=0;i<4;i++){
            int rand=(int) (Math.random() *36);
            rands[i]=chars.charAt(rand);
        }

        /*
         * 产生图像
         * 画背景
         */
        g.setColor(new Color(0xDCDCDC));
        g.fillRect(0, 0, width, height);

        /*
         * 随机产生120个干扰点
         */

        for(int i=0;i<120;i++){
            int x=(int)(Math.random()*width);
            int y=(int)(Math.random()*height);
            int red=(int)(Math.random()*255);
            int green=(int)(Math.random()*255);
            int blue=(int)(Math.random()*255);
            g.setColor(new Color(red,green,blue));
            g.drawOval(x, y, 1, 0);
        }
        g.setColor(Color.BLACK);
        g.setFont(new Font(null, Font.ITALIC|Font.BOLD,18));

        //在不同高度输出验证码的不同字符
        g.drawString(""+rands[0], 1, 17);
        g.drawString(""+rands[1], 16, 15);
        g.drawString(""+rands[2], 31, 18);
        g.drawString(""+rands[3], 46, 16);
        g.dispose();

        //将图像传到客户端
        ServletOutputStream sos=response.getOutputStream();
        ByteArrayOutputStream baos=new ByteArrayOutputStream();
        ImageIO.write(image, "JPEG", baos);
        byte[] buffer=baos.toByteArray();
        response.setContentLength(buffer.length);
        sos.write(buffer);
        baos.close();
        sos.close();
        session.setAttribute(codeName, new String(rands));
    }
    private  static boolean checkName(String name){
        boolean result=codeNameSet.contains(name)?true:false;
        return result;
    }

}
